{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 2,
   "metadata": {},
   "outputs": [
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "c:\\ProgramData\\miniconda3\\envs\\danwen_rag\\Lib\\site-packages\\tqdm\\auto.py:21: TqdmWarning: IProgress not found. Please update jupyter and ipywidgets. See https://ipywidgets.readthedocs.io/en/stable/user_install.html\n",
      "  from .autonotebook import tqdm as notebook_tqdm\n"
     ]
    }
   ],
   "source": [
    "from llama_index.core import VectorStoreIndex, SimpleDirectoryReader, Settings, StorageContext, load_index_from_storage\n",
    "from llama_index.core.chat_engine.types import ChatMode\n",
    "from llama_index.core.memory import ChatMemoryBuffer\n",
    "from llms import gangtise_llm\n",
    "from embeddings import embed_model_local_bge_small\n",
    "Settings.llm = gangtise_llm()\n",
    "Settings.embed_model = embed_model_local_bge_small()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 6,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "' \\n \\n轻工制造 / 行业投资策略周报 / 2024.10.01 \\n请阅读最后一页的重要声明！ \\n \\n \\n政策提振家装需求，关注相关家居转债  \\n证券研究报告 \\n投资评级:看好(维持) \\n最近 12 月市场表现 \\n \\n   \\n分析师 吕明璋 \\nSAC 证书编号：S0160523030001 \\nlvmz@ctsec.com \\n \\n联系人 何栋 \\nhedong@ctsec.com \\n \\n  \\n \\n相关报告 \\n1. 《一线城市优化限购政策，全面促进\\n地产止跌回稳》   2024-09-30 \\n2. 《政策提振信心，龙头预期改善》    \\n2024-09-27 \\n3. 《促进房地产市场止跌回稳，期待家\\n居需求释放》   2024-09-26   \\n \\n \\n  \\n核心观点    \\n\\uf076 促消费政策效果显现，家居家装销量大幅增长。据发改委消息，支持消费\\n品以旧换新的 1500 亿元超长期特别国债资金已于 8 月初全部下达到地方，政\\n策实施以来家居家装消费品销量实现大幅增长，4-8 月份，家具、卫生洁具零\\n售业销售收入同比分别增长 8.9%和 12%，增速较 3 月份分别提高了 8.3 个百\\n分点和 22.6 个百分点。目前，安徽、湖北、河北、云南等多地已经启动补贴\\n申领核销工作，加快释放家居消费潜力。伴随“金九银十” 、 “双十一”等重要\\n消费节点到来，政策效应有望持续释放。建议关注定制家居转债：欧22 转债、\\n金 23 转债、好客转债、江山转债等；卫浴制品转债：松霖转债；瓷砖地板转\\n债：蒙娜转债。 \\n\\uf076 本周转债行情表现：本周（2024.9.23-2024.9.27），中证转债指数涨 5.22%，\\n上证转债指数涨 4.69%，深证转债指数 涨 6.17%，万得可转债等权指数 涨\\n6.11%， 万得可转债加权指数涨 5.31%。轻工制造板块中，艾录转债 （+16.67%）、\\n鹰 19 转债（+12.70%）、帝欧转债（+12.69%）涨幅居前。 \\n\\uf076 本周转债条款信息：1）荣23转债：公司实际控制人冯荣华、张云芳夫妇于\\n2024 年 5 月 21 日至 2024 年 9 月 25 日，通过集中竞价交易方式合计减持“荣\\n23 转债”62.21 万张，占可转债发行总量的 10.80%。2）山鹰转债：公司拟下\\n修可转债转股价格，“山鹰转债”和“鹰 19 转债”于 2024 年 9 月 27 日暂停转\\n股，自 2024 年 9 月 30 日起，“山鹰转债”转股价格由 2.25 元/股调整为 1.54 元\\n/股；“鹰 19 转债”转股价格由 2.25 元/股调整为 2.24 元/股，并同日恢复转股。 \\n\\uf076 本周转债正股信息：1）东峰集团：公司股东东捷控股于 2024 年 7 月 8 日至\\n2024 年 9 月 24 日期间通过上海证券交易所集中竞价方式减持公司股份\\n1842.96 万股（约占公司总股本的 1%） 、通过上海证券交易所大宗交易方式减\\n持公司股份 3685.92 万股（约占公司总股本的2%） 。上述减持实施后，东捷控\\n股直接持有公司股份 4696.69 万股，占公司总股本的2.55%。2）山鹰国际：公\\n司及其下属子公司拟通过全国碳排放权交易系统以大宗协议、挂牌交易等多\\n种转让方式出售碳排放配额约 100 万吨，出售价格不低于 90 元/吨（含税） ，\\n交易总金额约 9000 万元。截至本公告披露日，公司已通过全国碳排放权交易\\n系统以大宗协议转让的方式出售 2019 年、2020 年碳排放配额 40.37 万吨，成\\n交均价 101 元/吨，交易总金额 4077.38 万元（含税） 。 \\n\\uf076 风险提示：地产销售持续下滑；原材料价格持续上涨；下游需求不及预期 \\n   \\n-32%\\n-23%\\n-15%\\n-7%\\n1%\\n9%\\n轻工制造 沪深300\\n'"
      ]
     },
     "execution_count": 6,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "data = SimpleDirectoryReader(input_dir=\"data\").load_data()\n",
    "type(data)\n",
    "data[0].text"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "index = VectorStoreIndex.from_documents(data, show_progress=True)\n"
   ]
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "danwen_rag",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.11.11"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}
